source /home/work_nfs7/xlgeng/.bashrc
dir=../../../../checkpoint/wenet_gxl_salmonn/salmonn_v15/5_epoch
mkdir -p $dir
train_config=conf/train_salmonn_v15.yaml
test_data_dir="/home/work_nfs8/xlgeng/new_workspace/wenet_gxl_salmonn/examples/aishell/salmonn/data_list/kespeech"
test_sets="Beijing Ji-Lu Jiang-Huai Jiao-Liao Lan-Yin Northeastern Southwestern Zhongyuan"
decode_checkpoint_name="step_44268"
decode_checkpoint=$dir/"${decode_checkpoint_name}".pt

stage=5
stop_stage=5


echo "final checkpoint is $decode_checkpoint"
test_sets=$(echo "$test_sets" | sed 's/ /--/g')
gxl_num="002"

prompt_types_1="repeat_prompt common_prompt"
prompt_types_2="long_prompt short_prompt"
prompt_types_3="encourage_prompt messy_prompt"
prompt_types_1="common_prompt"
prompt_types_2="short_prompt"
prompt_types_3="messy_prompt"
prompt_types_1=$(echo "$prompt_types_1" | sed 's/ /--/g')
prompt_types_2=$(echo "$prompt_types_2" | sed 's/ /--/g')
prompt_types_3=$(echo "$prompt_types_3" | sed 's/ /--/g')

my_named_nohup "${gxl_num}_kespeech_01" ./recognize_kespeech.sh --gpu_id 0 --dir $dir --train_config "$train_config" --decode_checkpoint "$decode_checkpoint" --decode_checkpoint_name "$decode_checkpoint_name" --stage $stage --stop_stage $stop_stage --test_data_dir "$test_data_dir" --prompt_types "$prompt_types_1" --test_sets "$test_sets"
my_named_nohup "${gxl_num}_kespeech_02" ./recognize_kespeech.sh --gpu_id 1 --dir $dir --train_config "$train_config" --decode_checkpoint "$decode_checkpoint" --decode_checkpoint_name "$decode_checkpoint_name" --stage $stage --stop_stage $stop_stage --test_data_dir "$test_data_dir" --prompt_types "$prompt_types_2" --test_sets "$test_sets"
my_named_nohup "${gxl_num}_kespeech_03" ./recognize_kespeech.sh --gpu_id 2 --dir $dir --train_config "$train_config" --decode_checkpoint "$decode_checkpoint" --decode_checkpoint_name "$decode_checkpoint_name" --stage $stage --stop_stage $stop_stage --test_data_dir "$test_data_dir" --prompt_types "$prompt_types_3" --test_sets "$test_sets"

